[리트코드]_15.3Sum
💯 solving-algo | February 18, 2021
문제 주소
높이를 입력받아 비 온 후 얼마나 많은 물이 쌓일 수 있는지 계산하는 문제 https://leetcode.com/problems/check-if-a-number-is-majority-element-in-a-sorted-array/
첫 번째 풀이
def threeSum(nums):
result = []
nums.sort()
# 브루투 포스 n^3 반복
for i in range(len(nums) - 2):
# 중복된 값 건너뛰기
if i > 0 and nums[i] == nums[i - 1]:
continue
for j in range(i + 1, len(nums) - 1):
if j > i + 1 and nums[j] == nums[j - 1]:
continue
for k in range(j + 1, len(nums)):
if k > j + 1 and nums[k] == nums[k -1]:
continue
if nums[i] + nums[j] + nums[k] == 0:
result.append((nums[i], nums[j], nums[k]))
return result
print(threeSum([-1, 0, 1, 2, -1, -4]))처음에 브루트 포스로 풀었더니 시간 초과가 났다. 그럴만한게 O(n^3)의 시간 복잡도를 가지고 있다.
두 번째 풀이
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
results = []
nums.sort()
for i in range(len(nums) - 2):
if i > 0 and nums[i] == nums[i -1]:
continue
left, right = i + 1, len(nums) - 1
# 간격을 좁혀가며 합 sum 계산
while left < right:
sum = nums[i] + nums[left] + nums[right]
if sum < 0:
left += 1
elif sum > 0:
right -= 1
else:
results.append((nums[i], nums[left], nums[right]))
while left < right and nums[left] == nums[left + 1]:
left += 1
while left < right and nums[right] == nums[right - 1]:
right -= 1
left += 1
right -= 1
return results풀이방법
투 포인터를 이용해 간극을 좁혀가며 풀었다. 앞에 중복되는 값 계산은 continue 처리했으며, sum이 0일때만 append 이후 left, right 값 증가/감소 처리를 하여 0을 도출해 냈다. 시간 복잡도는 O(n^2)이다.
